package com.device.common.datasource;

/**
 * @author lgb
 * @说明 Copyright (C) 深圳工勘科技 数据源切换处理
 **/
public class DataSourceContextHolder { 
	// 存放当前线程使用的数据源类型
	private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();

	// 设置数据源
	public static void setDataSource(String type) {
		contextHolder.set(type);
	}

	// 获取数据源
	public static String getDataSource() {
		return contextHolder.get();
	}

	// 清除数据源
	public static void clearDataSource() {
		contextHolder.remove();
	}
}